package com.shfl.Util;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFRichTextString;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.HSSFColor;


public class ExcelUtil {

	@SuppressWarnings("unchecked")
	public static Map buildExcel(String tempPath, List<String> dataKey, List<String> dataTitle, List list, String fileName) {
		String extension = ".xls";
		OutputStream op = getOutputStream(tempPath + "/", fileName, extension);
		exportExcel(dataKey, dataTitle, list, op);
		exportExcel(dataKey, dataTitle, list, op);
		try {
			op.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		//FileUtil.rename(tempPath + fileType + "/", fileName, extension);
		@SuppressWarnings("rawtypes")
		Map rMap = new HashMap();
		rMap.put("fileName", fileName + extension);
		return rMap;
	}

	
	/**
     * 导出Excel
     * @param dataKey 每列对应的key名
     * @param dataTitle 表格第一行中文
     * @param data 需要显示的数据集合
     * @param out 与输出设备关联的流对象，可以将EXCEL文档导出到本地文件或者网络中
     */
	public static void exportExcel(List dataKey, List dataTitle, List data, OutputStream out) {
        // 声明一个工作薄
        HSSFWorkbook workbook = new HSSFWorkbook();
        // 生成一个表格
        String title = "数据";
        HSSFSheet sheet = workbook.createSheet(title);
        // 设置表格默认列宽度为15个字节
        sheet.setDefaultColumnWidth(15);
        // 生成一个样式
        HSSFCellStyle style = workbook.createCellStyle();
        // 设置这些样式
        style.setFillForegroundColor(HSSFColor.WHITE.index);
        style.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
        style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        style.setBorderRight(HSSFCellStyle.BORDER_THIN);
        style.setBorderTop(HSSFCellStyle.BORDER_THIN);
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        // 生成一个字体
        HSSFFont font = workbook.createFont();
//        font.setColor(HSSFColor.VIOLET.index)
        font.setFontHeightInPoints((short) 12);
        font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
        // 把字体应用到当前的样式
        style.setFont(font);
        // 生成并设置另一个样式
        HSSFCellStyle style2 = workbook.createCellStyle();
        style2.setFillForegroundColor(HSSFColor.WHITE.index);
        style2.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
        style2.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        style2.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        style2.setBorderRight(HSSFCellStyle.BORDER_THIN);
        style2.setBorderTop(HSSFCellStyle.BORDER_THIN);
        style2.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        style2.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        // 生成另一个字体
        HSSFFont font2 = workbook.createFont();
        font2.setBoldweight(HSSFFont.BOLDWEIGHT_NORMAL);
        // 把字体应用到当前的样式
        style2.setFont(font2);

        // 声明一个画图的顶级管理器
//        HSSFPatriarch patriarch = sheet.createDrawingPatriarch()
//        // 定义注释的大小和位置,详见文档
//        HSSFComment comment = patriarch.createComment(new HSSFClientAnchor(0,
//                0, 0, 0, (short) 4, 2, (short) 6, 5))
//        // 设置注释内容
//        comment.setString(new HSSFRichTextString("可以在POI中添加注释！"))
//        // 设置注释作者，当鼠标移动到单元格上是可以在状态栏中看到该内容.
//        comment.setAuthor("leno")

        int keySize = dataKey.size();//导出Excel列的数量
        // 产生表格标题行
        HSSFRow row = sheet.createRow(0);
        for (int i = 0; i < keySize; i++) {
            HSSFCell cell = row.createCell(i);
            cell.setCellStyle(style);
            HSSFRichTextString text = new HSSFRichTextString(dataTitle.get(i).toString());
            cell.setCellValue(text);
        }
        // 遍历集合数据，产生数据行
        int rowSize = data.size();//数据行的数量
        for (int i = 0; i < rowSize; i++) {
            Map rowData = (Map) data.get(i);
            row = sheet.createRow(i + 2);
            for (int j = 0; j < keySize; j++) {
                HSSFCell cell = row.createCell(j);
                cell.setCellStyle(style2);
                String key = dataKey.get(j).toString();
                String textValue = getString(rowData.get(key));
                HSSFRichTextString text = new HSSFRichTextString(textValue);
                cell.setCellValue(text);
            }
        }
        try {
            workbook.write(out);
        }
        catch (IOException e) {
            e.printStackTrace();
        }
    }

    public  static String getString(Object obj) {
		if (obj == null) {
			return "";
		}
		return obj.toString().trim();
	}
    
    /**
	 * 获取文件字节流
	 * 
	 * @param path
	 *            路径
	 * @param fileName
	 *            文件名称
	 * @param extension
	 *            临时目录
	 * @return
	 */
	public static OutputStream getOutputStream(String path, String fileName, String extension) {
		File file = new File("/" + path.concat(fileName).concat(extension));
		FileOutputStream fos = null;
		try {
			fos = new FileOutputStream(file);
		} catch (FileNotFoundException e) {
			e.printStackTrace();
		}

		return fos;

	}
}
